# This Source Code Form is subject to the terms of the Mozilla Public
# License, v. 2.0. If a copy of the MPL was not distributed with this
# file, You can obtain one at https://mozilla.org/MPL/2.0/.  
# SPDX-License-Identifier: MPL-2.0

from __future__ import annotations
from typing import TYPE_CHECKING, List, Union
from PySide6.QtCore import Qt
from PySide6.QtGui import QPen, QCursor, QColor, QBrush
from PySide6.QtWidgets import (QGraphicsEllipseItem, QMenu,
                               QGraphicsSceneContextMenuEvent,
                               QGraphicsSceneMouseEvent)
from VeraGrid.Gui.gui_functions import add_menu_entry
from VeraGrid.Gui.Diagrams.MapWidget.Substation.node_template import NodeTemplate

from VeraGridEngine.Devices.types import INJECTION_DEVICE_TYPES

if TYPE_CHECKING:  # Only imports the below statements during type checking
    from VeraGrid.Gui.Diagrams.MapWidget.grid_map_widget import GridMapWidget


class MapInjectionTemplateGraphicItem(NodeTemplate, QGraphicsEllipseItem):
    """
    InjectionTemplateGraphicItem
    """

    def __init__(self,
                 api_object: INJECTION_DEVICE_TYPES,
                 editor: GridMapWidget,
                 lat: float,
                 lon: float,
                 size: float = 0.8,
                 draw_labels: bool = True):
        """

        :param api_object:
        :param editor:
        :param lat:
        :param lon:
        :param size:
        :param draw_labels:
        """
        # Correct way to call multiple inheritance
        super().__init__()

        # Explicitly call QGraphicsRectItem initialization
        QGraphicsEllipseItem.__init__(self)

        # Explicitly call NodeTemplate initialization
        NodeTemplate.__init__(self,
                              api_object=api_object,
                              editor=editor,
                              draw_labels=draw_labels,
                              lat=lat,
                              lon=lon)

        r2 = size / 2.0
        x, y = editor.to_x_y(lat=lat, lon=lon)  # upper left corner
        self.size = size
        self.setRect(x - r2, y - r2, self.size, self.size)

        # Properties of the container:
        self.setFlags(self.GraphicsItemFlag.ItemIsSelectable | self.GraphicsItemFlag.ItemIsMovable)
        self.setCursor(QCursor(Qt.CursorShape.PointingHandCursor))

        self.set_api_object_color()

    @property
    def api_object(self) -> INJECTION_DEVICE_TYPES:
        """

        :return:
        """
        return self._api_object

    @property
    def editor(self) -> GridMapWidget:
        """

        :return:
        """
        return self._editor

    def set_api_object_color(self) -> QColor:
        """
        Sets whatever colour is in the API object property
        """

        color = QColor(self.api_object.color)
        color.setAlpha(128)
        self.color = color

        brush = QBrush(color)
        self.setBrush(brush)

        pen = self.pen()
        pen.setColor(color)
        pen.setWidth(0)  # set the pen width to 0
        self.setPen(pen)

        return color

    def update_position_at_the_diagram(self) -> None:
        """
        Updates the element position in the diagram (to save)
        :return:
        """
        lat, long = self.editor.to_lat_lon(self.rect().x(), self.rect().y())

        self.lat = lat
        self.lon = long

        self.editor.update_diagram_element(device=self.api_object,
                                           latitude=lat,
                                           longitude=long,
                                           graphic_object=self)

    def set_size(self, r: float):
        """

        :param r: radius in pixels
        :return:
        """
        if r != self.size:
            rect = self.rect()
            rect.setWidth(r)
            rect.setHeight(r)

            # change the width and height while keeping the same center
            r2 = (self.size - r) / 2
            new_x = rect.x() + r2
            new_y = rect.y() + r2

            self.size = r

            # Set the new rectangle with the updated dimensions
            self.setRect(new_x, new_y, r, r)

    def recolour_mode(self):
        """
        Change the colour according to the system theme
        """
        super().recolour_mode()

        return self.set_api_object_color()

    def delete(self):
        """
        Remove this element
        @return:
        """

        deleted, delete_from_db_final = self.editor.delete_with_dialogue(selected=[self], delete_from_db=False)

    def plot(self):
        """
        Plot API objects profiles
        """
        # time series object from the last simulation
        ts = self.editor.circuit.time_profile

        # plot the profiles
        self.api_object.plot_profiles(time=ts)

    def mousePressEvent(self, event: QGraphicsSceneMouseEvent):
        """
        mouse press: display the editor
        :param event:
        :return:
        """
        super().mousePressEvent(event)
        self._editor.set_editor_model(api_object=self.api_object)

    def mouseReleaseEvent(self, event: QGraphicsSceneMouseEvent):
        """
        Event handler for mouse release events.
        """
        super().mouseReleaseEvent(event)
        # self.editor.disableMove = True
        self.update_position_at_the_diagram()  # always update

    def mouseDoubleClickEvent(self, event, /):
        """

        :param event:
        :return:
        """
        super().mouseDoubleClickEvent(event)
        self.set_api_object_color()

    def get_base_context_menu(self) -> QMenu:
        """
        Generate the base menu for injections
        :return:
        """
        menu = QMenu()

        add_menu_entry(menu=menu,
                       text="Plot profiles",
                       function_ptr=self.plot,
                       icon_path=":/Icons/icons/plot.png")

        add_menu_entry(menu=menu,
                       text="Consolidate coordinates",
                       function_ptr=self.consolidate_coordinates,
                       icon_path=":/Icons/icons/delete_schematic.png")

        menu.addSeparator()

        add_menu_entry(menu=menu,
                       text="Delete",
                       function_ptr=self.delete,
                       icon_path=":/Icons/icons/delete_schematic.png")

        return menu

    def contextMenuEvent(self, event: QGraphicsSceneContextMenuEvent):
        """
        Display context menu
        @param event:
        @return:
        """
        menu = self.get_base_context_menu()

        menu.exec(event.screenPos())

    def consolidate_coordinates(self):
        """
        Consolidate coordinates in to the DB
        """
        lat, long = self.editor.to_lat_lon(self.rect().x(), self.rect().y())
        self.api_object.latitude = lat
        self.api_object.longitude = long
        self.editor.gui.show_info_toast("Coordinates consolidated!")
